THEANO_FLAGS='device=gpu2, floatX=float32' python -mipdb main.py --state_dim 700 --time_length 150 --mini_batch_size 128 --layers 4 --save_path '/data/lisatmp3/zablocki/4XLSTM_700Units_ADAM' --load_path '/data/lisatmp3/zablocki/4XLSTM_700Units_ADAM/best_jun.npz' --dataset 'wikipedia_junyoung' --monitoring_freq 1 --skip_connections --interactive_mode